package Tree;

/**
 * @Classname JZ55平衡二叉树
 * @Description TODO
 * @Date 2023/2/23 16:51
 * @Created by xjl
 */
public class JZ55平衡二叉树 {
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

    public boolean isBalanced(TreeNode root) {
        if (root == null) {
            return true;
        } else {
            // 要求这个左子树的和右子树都要小于这个，同时要求这个树本身的高度差不能超过1
            return Math.abs(height(root.left) - height(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right);
        }
    }

    public int height(TreeNode root) {
        if (root == null) {
            return 0;
        } else {
            // 返回的是这个树的高度
            return Math.max(height(root.left), height(root.right)) + 1;
        }
    }
}
